package com.example.questionbase.handler;

import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import org.apache.ibatis.type.*;

import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Arrays;
import java.util.List;

/**
 * @Author: fan
 * @Description: TODO
 * @Date: 2024/03/01 14:36
 */
public class KeywordTypeHandler extends BaseTypeHandler<List<String>> {
    @Override
    public void setNonNullParameter(PreparedStatement ps, int i, List<String> parameter, JdbcType jdbcType) throws SQLException {
        // 遍历List类型的入参，拼装为String类型，使用Statement对象插入数据库
        StringBuffer sb = new StringBuffer();
        for (int j = 0; j < parameter.size(); j++) {
            if (j == parameter.size() - 1) {
                sb.append(parameter.get(j));
            } else {
                sb.append(parameter.get(j)).append(";");
            }
        }
        ps.setString(i, sb.toString());
    }

    @Override
    public List<String> getNullableResult(ResultSet rs, String columnName) throws SQLException {
        // 获取String类型的结果，使用","分割为List后返回
        String resultString = rs.getString(columnName);
        if (StringUtils.isNotBlank(resultString)) {
            return Arrays.asList(resultString.split(";"));
        }
        return null;
    }

    @Override
    public List<String> getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
        // 获取String类型的结果，使用","分割为List后返回
        String resultString = rs.getString(columnIndex);
        if (StringUtils.isNotBlank(resultString)) {
            return Arrays.asList(resultString.split(";"));
        }
        return null;
    }

    @Override
    public List<String> getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
        // 获取String类型的结果，使用","分割为List后返回
        String resultString = cs.getString(columnIndex);
        if (StringUtils.isNotBlank(resultString)) {
            return Arrays.asList(resultString.split(";"));
        }
        return null;
    }
}
